home *** CD-ROM | disk | FTP | other *** search
/ Network Supervisor's Toolkit / Network Supervisor's Toolkit.iso / netbios / nbsource / nbiosser
Text File  |  1996-07-10  |  9KB  |  244 lines
  1. bunfigs
  2. Figure 1.  Initialize NetBios Driver 
  3.  
  4.   /*--------------------------------------------------------* 
  5.    * Initialize NetBios driver by getting its name, through * 
  6.    * NetBiosEnum() and opening it with NetBiosOpen()        * 
  7.    *--------------------------------------------------------*/ 
  8.   InitializeNetbios(WORD *devHandle) 
  9.   { 
  10.     int ccode; 
  11.     struct netbios_info_1 NB_INFO; 
  12.     unsigned int totEntries, numEntries; 
  13.  
  14.     if (ccode = NetBiosEnum((char far *)"", 1,  
  15.                 (char far *)&NB_INFO, 
  16.                 (unsigned)sizeof(NB_INFO),  
  17.                 (unsigned short far *)&numEntries, 
  18.                 (unsigned short far *)&totEntries)) 
  19.         printf("Error occurred during NetBiosEnum: %x\n", ccode); 
  20.     if (ccode = NetBiosOpen((char far *)NB_INFO.nb1_driver_name,  
  21.                (char far *)0, (unsigned)NB_REGULAR, 
  22.                (unsigned short far *)devHandle)) 
  23.         printf("Error occurred during NetBiosOpen: %x.\n", ccode); 
  24.   }  Figure 2.  Adding a name via the thread AddServerName. 
  25.  
  26.  LONG    wait_sem_val = 0; 
  27.  HSEM    wait_sem = &wait_sem_val; 
  28.   
  29.  main() 
  30.  { 
  31.  . . . . 
  32.     DosSemSet(wait_sem); 
  33.     stackPtr = (WORD far *) malloc(1024) + (1024 / sizeof(WORD)); 
  34.     ccode = DosCreateThread(AddServerName, &addNameThreadID, (char far *)stackPtr); 
  35.     if (ccode) 
  36.         printf("DosCreateThread Error : %d\n", ccode); 
  37.         . . . .  
  38.     DosSemWait(wait_sem, -1L);        /* wait for the name to be added */ 
  39. } 
  40.  
  41.   /*------------------------------------------------------------------------* 
  42.    * AddServerName adds the servers name, clearing the wait_sem when done.  * 
  43.    * It kills itself upon termination, if an error occurs it kills everyone * 
  44.    *------------------------------------------------------------------------*/ 
  45.   void far AddServerName() 
  46.   { 
  47.    if (!AddName(SERVER_NAME, devHandle)) { 
  48.         DeleteName(SERVER_NAME, devHandle); 
  49.         if (!AddName(SERVER_NAME, devHandle)) { 
  50.             DosSemClear(wait_sem); 
  51.             printf("\nError in NetBios, could't add name"); 
  52.             DosExit (1, 0); 
  53.         } 
  54.     } 
  55.     DosSemClear(wait_sem); 
  56.     DosExit (0, 0); 
  57.   } 
  58.  
  59.   /*------------------------------------------------*/ 
  60.   AddName(char *name, WORD devHandle) 
  61.   { 
  62.     NCB addNCB; 
  63.  
  64.     memset((char *)&addNCB, '\0', 64); 
  65.     addNCB.ncb_command = NCBADDNAME; 
  66.     addNCB.ncb_lana_num = 0x00; 
  67.     memcpy(addNCB.ncb_name, name, strlen(name)); 
  68.     if (NetBiosSubmit(devHandle, SINGLE_RETRY, &addNCB)) 
  69.         return(FALSE); 
  70.     else 
  71.         return (TRUE); 
  72. }  Figure 3.  Cloned communications function. 
  73.  
  74. /*----------------------------------------------------------------------* 
  75.  * Receiver is the server side of the IPC connection.  It makes, reads, * 
  76.  * writes, and closes the pipe until 'down' is entered.                 * 
  77.  *----------------------------------------------------------------------*/ 
  78. void far Receiver(int clientNum) 
  79. { 
  80.     COMMBUFFER commBuffer; 
  81.     int ccode; 
  82.  
  83.     DosSemWait(wait_sem, -1L);          /* wait for name to be added */ 
  84.     do { 
  85.         ListenForClient(&sessionNum[clientNum]); 
  86.         do { 
  87.             DoReceive((char *)&commBuffer, sizeof(COMMBUFFER), 
  88.                    sessionNum[clientNum], devHandle, &ncb[clientNum]); 
  89.             if (!TLAServiceRequest(&commBuffer)) 
  90.                 strcpy(commBuffer.record.explain, 
  91.                               "The TLA requested could not be found."); 
  92.             DoSend((char *)&commBuffer, sizeof(COMMBUFFER), 
  93.                    sessionNum[clientNum], devHandle, &ncb[clientNum]); 
  94.             if (commBuffer.lastOne) 
  95.                 break; 
  96.         } while (TRUE); 
  97.         /* client hangs up, so we don't have to */ 
  98.         sessionNum[clientNum] = 0; 
  99.     } while (TRUE); 
  100. } 
  101. Figure 4.  NetBios Listen command. 
  102.  
  103. /*----------------------------------------------------------------------* 
  104.  * Listen for someone to call us, don't return until we've been called  * 
  105.  *----------------------------------------------------------------------*/ 
  106. ListenForClient(int *sessionNum) 
  107. { 
  108.     int ccode; 
  109.     NCB listenNCB; 
  110.  
  111.     memset((char *)&listenNCB, '\0', 64); 
  112.     listenNCB.ncb_command = NCBLISTEN | ASYNCH; 
  113.     strcpy(listenNCB.ncb_callname, "*"); 
  114.     strcpy(listenNCB.ncb_name, SERVER_NAME); 
  115.     listenNCB.ncb_lana_num = 0x00; 
  116.     listenNCB.ncb_rto = 0x30; 
  117.     listenNCB.ncb_sto = 0x10; 
  118.  
  119.     if (ccode = NetBiosSubmit(devHandle, SINGLE_RETRY, &listenNCB)) 
  120.         printf("NCB Submit error. ccode = %x.\n", ccode); 
  121.     while (listenNCB.ncb_cmd_cplt == 0xFF); 
  122.     if (listenNCB.ncb_cmd_cplt == 0) 
  123.         *sessionNum = listenNCB.ncb_lsn; 
  124. } 
  125. Figure 5.  NetBios commands Send and Receive. 
  126.  
  127. /*--------------------------------------------------------*/ 
  128. DoSend(char *buffer, int size, int sessionNum,  
  129.        WORD devHandle, NCB *ncb) 
  130. { 
  131.     int ccode; 
  132.     NCB sendNCB; 
  133.  
  134.     memset( (char *)ncb, '\0', 64 ); 
  135.     ncb->ncb_command = NCBSEND; 
  136.     ncb->ncb_lana_num = 0; 
  137.     ncb->ncb_lsn = sessionNum; 
  138.     ncb->ncb_length = size; 
  139.     ncb->ncb_buffer = buffer; 
  140.     if (ccode = NetBiosSubmit(devHandle, SINGLE_RETRY, ncb)) { 
  141.         printf("Error sending, submit error: %x\n", ccode); 
  142.         return (FALSE); 
  143.       } 
  144.     return (TRUE); 
  145. } 
  146.  
  147. /*--------------------------------------------------------*/ 
  148. DoReceive(char *buffer, int length, int sessionNum,  
  149.           WORD devHandle, NCB *ncb) 
  150. { 
  151.     int ccode; 
  152.  
  153.     memset( (char *)ncb, '\0', 64 ); 
  154.     ncb->ncb_command = NCBRECV; 
  155.     ncb->ncb_lana_num = 0; 
  156.     ncb->ncb_lsn = sessionNum; 
  157.     ncb->ncb_length = length; 
  158.     ncb->ncb_buffer = buffer; 
  159.     if (ccode = NetBiosSubmit(devHandle, SINGLE_RETRY, ncb)) { 
  160.         printf("\nError receiving, submit error: %x\n", ccode); 
  161.         return (FALSE); 
  162.     } 
  163.     return (TRUE); 
  164. } Figure 6.  Close NetBios down. 
  165.  
  166.  /*--------------------------------------------------------* 
  167.  * CleanUp performs the exiting procedures, disconnecting * 
  168.  * and closing open sessions.                             *                                                   
  169.        * 
  170.  *--------------------------------------------------------*/ 
  171. CleanUp() 
  172. { 
  173.     int ccode; 
  174.     int i; 
  175.     char response[20]; 
  176.  
  177.     /*  wait until 'down' is indicated  */ 
  178.     do { 
  179.         scanf("%s", response); 
  180.     } while (strcmp(response, "down") != 0); 
  181.     DosEnterCritSec(); 
  182.     TLACloseIndex(); 
  183.     for (i = 0; i < CLIENT_NUM; i++) { 
  184.         if (sessionNum[i] != 0) { 
  185.             CancelNCB(ncb[i]);     /* cancel outstanding NCB */ 
  186.             DoHangUp(sessionNum[i]); 
  187.         } 
  188.     } 
  189.     DeleteName(SERVER_NAME, devHandle); 
  190.     if (ccode = NetBiosClose(devHandle, (int)0)) 
  191.         printf("Error occurred during NetBiosClose. ccode = %x\n"); 
  192.     DosExit(1, 0); 
  193.     DosExitCritSec(); 
  194. } 
  195.  
  196. Figure 7.  NetBios Call command 
  197.  
  198. /*------------------------------------------------------*/ 
  199. DoCall(char *callName, WORD devHandle, int *sessionNum,  
  200.        char *yourName) 
  201. { 
  202.     NCB callNCB; 
  203.     int ccode; 
  204.  
  205.     memset((char *)&callNCB, '\0', 64); 
  206.     callNCB.ncb_command = NCBCALL; 
  207.     callNCB.ncb_lana_num = 0; 
  208.     callNCB.ncb_rto = 0x10; 
  209.     callNCB.ncb_sto = 0x10; 
  210.     memcpy(callNCB.ncb_callname, callName, strlen(callName)); 
  211.     memcpy(callNCB.ncb_name, yourName, strlen(yourName)); 
  212.     if (ccode = NetBiosSubmit(devHandle, SINGLE_RETRY, &callNCB)) 
  213.         printf("Error calling. NCB Submit error # %x\n", ccode); 
  214.  
  215.    if (callNCB.ncb_cmd_cplt != 0) { 
  216.        printf("\tError in CALL command: %x\n", callNCB.ncb_cmd_cplt); 
  217.        return (FALSE);  
  218.    } 
  219.    *sessionNum = callNCB.ncb_lsn; 
  220.    return (TRUE); 
  221. }  Figure 8.  Commbuffer description. 
  222.  
  223. /* record structure, external data file is a list of these */ 
  224. /* structures, does not have to be sorted                  */ 
  225. typedef struct { 
  226.     char tla[10];      /* the tla of record */ 
  227.     USHORT offset;     /* offset of next record with this TLA */ 
  228.     char expanded[80]; /* the expanded tla */ 
  229.     char explain[160]; /* an explanation of the tla */ 
  230. } RECORD; 
  231.  
  232. /* communication buffer passed into the server module */ 
  233. typedef struct { 
  234.     int type;                /* type of communication - see types below */ 
  235.     RECORD record;           /* information offered */ 
  236.     int lastOne;             /* are there any more requests on this session */ 
  237. } COMMBUFFER;  
  238.  
  239. /* Types of requests that come to server */ 
  240. #define ADD_TO_LIST       0x0001    /* add new tla */ 
  241. #define QUERY             0x0002    /* query to get expanded string only */ 
  242. #define SORT              0x0004    /* sort file and create new index */ 
  243.  
  244.